Information-Gain Calibrated On-Policy Distillation for Deep Search Agents
基于信息增益校准的在线策略蒸馏用于深度搜索智能体
🧭 On-policy distillation · 信息增益校准 ★ 深度解读
The Chinese University of Hong Kong, Shenzhen · Renmin University of China · Beijing University of Posts and Telecommunications · Beijing Academy of Artificial Intelligence (BAAI)
提出 IG-OPD,用检索后信息增益校准 OPD 的预算与方向先验,修正证据盲监督
📌 工作参考价值(4 点)
- 多轮检索/会话 agent 蒸馏时,teacher 对 query 的似然偏好不等于检索证据的实际效用;可引入 post-retrieval information gain(如文档是否包含答案、证据是否改变后续置信)作为校准信号,对 query/tool-call tokens 做奖励修正,避免好 query 被误罚。
- 按 turn 而不是按 token 长度分配监督权重:搜索/召回 turn 通常短但信息增益高,合成/回答 turn 长但增益低;ITC 的预算重分配可通过 token loss 乘 ratio 实现,无需改 trainer,工程成本低。
- 在电商搜索/推荐中,用户 query 改写、检索式推荐、多跳推理 agent 训练可借鉴 IQC 只修正 action-critical tokens(query 字符串、工具参数)而保持 reasoning tokens 不变;对第一跳正反馈、后续跳负反馈更可靠,可据此设计课程和 cosine ramp 避免早期对搜索行为的过度惩罚。
- 训练后模型推理无额外开销,但蒸馏时需额外做一次 teacher 对 gold answer 的评分,增加约 10% 训练时间/显存,适合离线训练阶段接受。